package com.ling.sbbyyds.common;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.experimental.Accessors;
import org.springframework.util.ObjectUtils;

import java.util.List;
import java.util.Set;

/**
 * 分页工具类
 *
 * @author ling
 */
@Data
@Accessors(chain = true)
public class PageUtils {

    /**
     * 总记录数
     */
    private int totalCount;
    /**
     * 每页记录数
     */
    private int pageSize;
    /**
     * 总页数
     */
    private int totalPage;
    /**
     * 当前页数
     */
    private int currPage;
    /**
     * 列表数据
     */
    private List<?> list;


    /***
     * 分页
     * @param list
     * 列表数据
     * @param totalCount  总记录数
     * @param pageSize    每页记录数
     * @param currPage    当前页数
     */
    public PageUtils(List<?> list, int totalCount, int pageSize, int currPage) {
        this.list = list;
        this.totalCount = totalCount;
        this.pageSize = pageSize;
        this.currPage = currPage;
        this.totalPage = (int) Math.ceil((double) totalCount / pageSize);
    }

    public PageUtils(IPage<?> page) {
        this.list = page.getRecords();
        this.totalCount = (int) page.getTotal();
        this.pageSize = (int) page.getSize();
        this.currPage = (int) page.getCurrent();
        this.totalPage = (int) page.getPages();
    }

    /**
     * 初始化分页参数
     *
     * @param currentPage 当前页
     * @param limit       一页显示数据量
     * @return 分页配置
     */
    public static Page initPage(Integer currentPage, Integer limit) {
        currentPage = ObjectUtils.isEmpty(currentPage) ? 1 : currentPage;
        limit = ObjectUtils.isEmpty(limit) ? 10 : limit;
        return new Page<>(currentPage, limit);
    }

    /**
     * 分页
     */
    public static PageUtils beautifyField(IPage<?> page) {
        return new PageUtils(page);
    }
}
